from gmssl import sm4

def sm4_encode(key, data):
    """
    国密sm4加密
    :param key: 密钥
    :param data: 原始数据
    :return: 密文hex
    """
    sm4Alg = sm4.CryptSM4()  # 实例化sm4
    sm4Alg.set_key(key.encode(), sm4.SM4_ENCRYPT)  # 设置密钥
    enRes = sm4Alg.crypt_ecb(data.encode())  # 开始加密,bytes类型，ecb模式
    return enRes.hex()  # 返回十六进制值

def sm4_decode(key, data):
    """
    国密sm4解密
    :param key: 密钥
    :param data: 密文数据
    :return: 明文bytes
    """
    sm4Alg = sm4.CryptSM4()  # 实例化sm4
    sm4Alg.set_key(key.encode(), sm4.SM4_DECRYPT)  # 设置密钥
    deRes = sm4Alg.crypt_ecb(bytes.fromhex(data))  # 开始解密。十六进制类型,ecb模式
    return deRes  # 返回解密后的字节数据

#测试函数
def test():
    key = "E1A90FB64DDE12AE"
    strData = "10"
    enHexRes = sm4_encode(key,strData)
    print("加密后的密文:",enHexRes)
    deByteRes = sm4_decode(key,enHexRes)
    print("解密后的明文:",deByteRes.decode())

# main
if __name__ == '__main__':
    print("main begin")
    test()